1   /*
2    * Copyright (C) 2011 The Guava Authors
3    *
4    * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
5    * in compliance with the License. You may obtain a copy of the License at
6    *
7    * http://www.apache.org/licenses/LICENSE-2.0
8    *
9    * Unless required by applicable law or agreed to in writing, software distributed under the License
10   * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
11   * or implied. See the License for the specific language governing permissions and limitations under
12   * the License.
13   */
14  
15  package com.google.common.cache;
16  
17  /**
18   * Utility {@link Weigher} implementations intended for use in testing.
19   *
20   * @author Charles Fry
21   */
22  public class TestingWeighers {
23  
24    /**
25     * Returns a {@link Weigher} that returns the given {@code constant} for every request.
26     */
27    static Weigher<Object, Object> constantWeigher(int constant) {
28      return new ConstantWeigher(constant);
29    }
30  
31    /**
32     * Returns a {@link Weigher} that uses the integer key as the weight.
33     */
34    static Weigher<Integer, Object> intKeyWeigher() {
35      return new IntKeyWeigher();
36    }
37  
38    /**
39     * Returns a {@link Weigher} that uses the integer value as the weight.
40     */
41    static Weigher<Object, Integer> intValueWeigher() {
42      return new IntValueWeigher();
43    }
44  
45    static final class ConstantWeigher implements Weigher<Object, Object> {
46      private final int constant;
47  
48      ConstantWeigher(int constant) {
49        this.constant = constant;
50      }
51  
52      @Override
53      public int weigh(Object key, Object value) {
54        return constant;
55      }
56    }
57  
58    static final class IntKeyWeigher implements Weigher<Integer, Object> {
59      @Override
60      public int weigh(Integer key, Object value) {
61        return key;
62      }
63    }
64  
65    static final class IntValueWeigher implements Weigher<Object, Integer> {
66      @Override
67      public int weigh(Object key, Integer value) {
68        return value;
69      }
70    }
71  
72  }